import {
  IAgent,
  IAgentConfig,
  ILLMConfig,
  IModelConfig,
  ISkill,
  IStatus,
  IWorkflow,
} from "./types";

export const getServerUrl = () => {
  return process.env.GATSBY_API_URL || "/api";
};

export function setCookie(name: string, value: any, days: number) {
  let expires = "";
  if (days) {
    const date = new Date();
    date.setTime(date.getTime() + days * 24 * 60 * 60 * 1000);
    expires = "; expires=" + date.toUTCString();
  }
  document.cookie = name + "=" + (value || "") + expires + "; path=/";
}

export function getCookie(name: string) {
  const nameEQ = name + "=";
  const ca = document.cookie.split(";");
  for (let i = 0; i < ca.length; i++) {
    let c = ca[i];
    while (c.charAt(0) == " ") c = c.substring(1, c.length);
    if (c.indexOf(nameEQ) == 0) return c.substring(nameEQ.length, c.length);
  }
  return null;
}
export function setLocalStorage(
  name: string,
  value: any,
  stringify: boolean = true
) {
  if (stringify) {
    localStorage.setItem(name, JSON.stringify(value));
  } else {
    localStorage.setItem(name, value);
  }
}

export function getLocalStorage(name: string, stringify: boolean = true): any {
  if (typeof window !== "undefined") {
    const value = localStorage.getItem(name);
    try {
      if (stringify) {
        return JSON.parse(value!);
      } else {
        return value;
      }
    } catch (e) {
      return null;
    }
  } else {
    return null;
  }
}

export function fetchJSON(
  url: string | URL,
  payload: any = {},
  onSuccess: (data: any) => void,
  onError: (error: IStatus) => void,
  onFinal: () => void = () => {}
) {
  return fetch(url, payload)
    .then(function (response) {
      if (response.status !== 200) {
        console.log(
          "Looks like there was a problem. Status Code: " + response.status,
          response
        );
        response.json().then(function (data) {
          console.log("Error data", data);
        });
        onError({
          status: false,
          message:
            "Connection error " + response.status + " " + response.statusText,
        });
        return;
      }
      return response.json().then(function (data) {
        onSuccess(data);
      });
    })
    .catch(function (err) {
      console.log("Fetch Error :-S", err);
      onError({
        status: false,
        message: `There was an error connecting to server. (${err}) `,
      });
    })
    .finally(() => {
      onFinal();
    });
}
export const capitalize = (s: string) => {
  if (typeof s !== "string") return "";
  return s.charAt(0).toUpperCase() + s.slice(1);
};

export function eraseCookie(name: string) {
  document.cookie = name + "=; Path=/; Expires=Thu, 01 Jan 1970 00:00:01 GMT;";
}

export function truncateText(text: string, length = 50) {
  if (text.length > length) {
    return text.substring(0, length) + " ...";
  }
  return text;
}

export const getCaretCoordinates = () => {
  let caretX, caretY;
  const selection = window.getSelection();
  if (selection && selection?.rangeCount !== 0) {
    const range = selection.getRangeAt(0).cloneRange();
    range.collapse(false);
    const rect = range.getClientRects()[0];
    if (rect) {
      caretX = rect.left;
      caretY = rect.top;
    }
  }
  return { caretX, caretY };
};

export const getPrefixSuffix = (container: any) => {
  let prefix = "";
  let suffix = "";
  if (window.getSelection) {
    const sel = window.getSelection();
    if (sel && sel.rangeCount > 0) {
      let range = sel.getRangeAt(0).cloneRange();
      range.collapse(true);
      range.setStart(container!, 0);
      prefix = range.toString();

      range = sel.getRangeAt(0).cloneRange();
      range.collapse(true);
      range.setEnd(container, container.childNodes.length);

      suffix = range.toString();
      console.log("prefix", prefix);
      console.log("suffix", suffix);
    }
  }
  return { prefix, suffix };
};

export const uid = () => {
  return Date.now().toString(36) + Math.random().toString(36).substr(2);
};

export const setCaretToEnd = (element: HTMLElement) => {
  const range = document.createRange();
  const selection = window.getSelection();
  range.selectNodeContents(element);
  range.collapse(false);
  selection?.removeAllRanges();
  selection?.addRange(range);
  element.focus();
};

// return a color between a start and end color using a percentage
export const ColorTween = (
  startColor: string,
  endColor: string,
  percent: number
) => {
  // example startColor = "#ff0000" endColor = "#0000ff" percent = 0.5
  const start = {
    r: parseInt(startColor.substring(1, 3), 16),
    g: parseInt(startColor.substring(3, 5), 16),
    b: parseInt(startColor.substring(5, 7), 16),
  };
  const end = {
    r: parseInt(endColor.substring(1, 3), 16),
    g: parseInt(endColor.substring(3, 5), 16),
    b: parseInt(endColor.substring(5, 7), 16),
  };
  const r = Math.floor(start.r + (end.r - start.r) * percent);
  const g = Math.floor(start.g + (end.g - start.g) * percent);
  const b = Math.floor(start.b + (end.b - start.b) * percent);
  return `rgb(${r}, ${g}, ${b})`;
};

export const guid = () => {
  var w = () => {
    return Math.floor((1 + Math.random()) * 0x10000)
      .toString(16)
      .substring(1);
  };
  return `${w()}${w()}-${w()}-${w()}-${w()}-${w()}${w()}${w()}`;
};

/**
 * Takes a string and returns the first n characters followed by asterisks.
 * @param {string} str - The string to obscure
 * @param {number} n - Number of characters to show before obscuring
 * @returns {string} The obscured string with first n characters in clear text
 */
export const obscureString = (str: string, n: number = 3) => {
  if (n < 0 || n > str.length) {
    console.log("n cannot be less than 0 or greater than the string length.");
    return str;
  }
  // First n characters in clear text
  var clearText = str.substring(0, n);
  // Remaining characters replaced with asterisks
  var obscured = clearText + "*".repeat(str.length - n);

  return obscured;
};

/**
 * Converts a number of seconds into a human-readable string representing the duration in days, hours, minutes, and seconds.
 * @param {number} seconds - The number of seconds to convert.
 * @returns {string} A well-formatted duration string.
 */
export const formatDuration = (seconds: number) => {
  const units = [
    { label: " day", seconds: 86400 },
    { label: " hr", seconds: 3600 },
    { label: " min", seconds: 60 },
    { label: " sec", seconds: 1 },
  ];

  let remainingSeconds = seconds;
  const parts = [];

  for (const { label, seconds: unitSeconds } of units) {
    const count = Math.floor(remainingSeconds / unitSeconds);
    if (count > 0) {
      parts.push(count + (count > 1 ? label + "s" : label));
      remainingSeconds -= count * unitSeconds;
    }
  }

  return parts.length > 0 ? parts.join(" ") : "0 sec";
};

export const sampleModelConfig = (modelType: string = "open_ai") => {
  const openaiConfig: IModelConfig = {
    model: "gpt-4-1106-preview",
    api_type: "open_ai",
    description: "OpenAI GPT-4 model",
  };
  const azureConfig: IModelConfig = {
    model: "gpt-4",
    api_type: "azure",
    api_version: "v1",
    base_url: "https://youazureendpoint.azure.com/",
    description: "Azure model",
  };

  const googleConfig: IModelConfig = {
    model: "gemini-1.0-pro",
    api_type: "google",
    description: "Google Gemini Model model",
  };

  switch (modelType) {
    case "open_ai":
      return openaiConfig;
    case "azure":
      return azureConfig;
    case "google":
      return googleConfig;
    default:
      return openaiConfig;
  }
};

export const getRandomIntFromDateAndSalt = (salt: number = 43444) => {
  const currentDate = new Date();
  const seed = currentDate.getTime() + salt;
  const randomValue = Math.sin(seed) * 10000;
  const randomInt = Math.floor(randomValue) % 100;
  return randomInt;
};

export const sampleAgentConfig = (agent_type: string = "assistant") => {
  const llm_config: ILLMConfig = {
    config_list: [],
    temperature: 0.1,
    timeout: 600,
    cache_seed: null,
    max_tokens: 1000,
  };

  const userProxyConfig: IAgentConfig = {
    name: "userproxy",
    human_input_mode: "NEVER",
    description: "User Proxy",
    max_consecutive_auto_reply: 25,
    system_message: "You are a helpful assistant.",
    default_auto_reply: "TERMINATE",
    llm_config: false,
    code_execution_config: "local",
  };
  const userProxyFlowSpec: IAgent = {
    type: "userproxy",
    config: userProxyConfig,
  };

  const assistantConfig: IAgentConfig = {
    name: "primary_assistant",
    description: "Primary Assistant",
    llm_config: llm_config,
    human_input_mode: "NEVER",
    max_consecutive_auto_reply: 25,
    code_execution_config: "none",
    system_message:
      "You are a helpful AI assistant. Solve tasks using your coding and language skills. In the following cases, suggest python code (in a python coding block) or shell script (in a sh coding block) for the user to execute. 1. When you need to collect info, use the code to output the info you need, for example, browse or search the web, download/read a file, print the content of a webpage or a file, get the current date/time, check the operating system. After sufficient info is printed and the task is ready to be solved based on your language skill, you can solve the task by yourself. 2. When you need to perform some task with code, use the code to perform the task and output the result. Finish the task smartly. Solve the task step by step if you need to. If a plan is not provided, explain your plan first. Be clear which step uses code, and which step uses your language skill. When using code, you must indicate the script type in the code block. The user cannot provide any other feedback or perform any other action beyond executing the code you suggest. The user can't modify your code. So do not suggest incomplete code which requires users to modify. Don't use a code block if it's not intended to be executed by the user. If you want the user to save the code in a file before executing it, put # filename: <filename> inside the code block as the first line. Don't include multiple code blocks in one response. Do not ask users to copy and paste the result. Instead, use 'print' function for the output when relevant. Check the execution result returned by the user. If the result indicates there is an error, fix the error and output the code again. Suggest the full code instead of partial code or code changes. If the error can't be fixed or if the task is not solved even after the code is executed successfully, analyze the problem, revisit your assumption, collect additional info you need, and think of a different approach to try. When you find an answer, verify the answer carefully. Include verifiable evidence in your response if possible. Reply 'TERMINATE' in the end when everything is done.",
  };

  const assistantFlowSpec: IAgent = {
    type: "assistant",
    config: assistantConfig,
  };

  const groupChatAssistantConfig = Object.assign(
    {
      admin_name: "groupchat_assistant",
      messages: [],
      max_round: 10,
      speaker_selection_method: "auto",
      allow_repeat_speaker: false,
    },
    assistantConfig
  );
  groupChatAssistantConfig.name = "groupchat_assistant";
  groupChatAssistantConfig.system_message =
    "You are a helpful assistant skilled at cordinating a group of other assistants to solve a task. ";
  groupChatAssistantConfig.description = "Group Chat Assistant";

  const groupChatFlowSpec: IAgent = {
    type: "groupchat",
    config: groupChatAssistantConfig,
  };

  if (agent_type === "userproxy") {
    return userProxyFlowSpec;
  } else if (agent_type === "assistant") {
    return assistantFlowSpec;
  } else if (agent_type === "groupchat") {
    return groupChatFlowSpec;
  } else {
    return assistantFlowSpec;
  }
};

export const sampleWorkflowConfig = (type = "twoagents") => {
  const llm_model_config: IModelConfig[] = [];

  const llm_config: ILLMConfig = {
    config_list: llm_model_config,
    temperature: 0.1,
    timeout: 600,
    cache_seed: null,
    max_tokens: 1000,
  };

  const userProxyConfig: IAgentConfig = {
    name: "userproxy",
    human_input_mode: "NEVER",
    max_consecutive_auto_reply: 15,
    system_message: "You are a helpful assistant.",
    default_auto_reply: "TERMINATE",
    llm_config: false,
    code_execution_config: "local",
  };
  const userProxyFlowSpec: IAgent = {
    type: "userproxy",
    config: userProxyConfig,
  };

  const assistantConfig: IAgentConfig = {
    name: "primary_assistant",
    llm_config: llm_config,
    human_input_mode: "NEVER",
    max_consecutive_auto_reply: 8,
    code_execution_config: "none",
    system_message:
      "You are a helpful AI assistant. Solve tasks using your coding and language skills. In the following cases, suggest python code (in a python coding block) or shell script (in a sh coding block) for the user to execute. 1. When you need to collect info, use the code to output the info you need, for example, browse or search the web, download/read a file, print the content of a webpage or a file, get the current date/time, check the operating system. After sufficient info is printed and the task is ready to be solved based on your language skill, you can solve the task by yourself. 2. When you need to perform some task with code, use the code to perform the task and output the result. Finish the task smartly. Solve the task step by step if you need to. If a plan is not provided, explain your plan first. Be clear which step uses code, and which step uses your language skill. When using code, you must indicate the script type in the code block. The user cannot provide any other feedback or perform any other action beyond executing the code you suggest. The user can't modify your code. So do not suggest incomplete code which requires users to modify. Don't use a code block if it's not intended to be executed by the user. If you want the user to save the code in a file before executing it, put # filename: <filename> inside the code block as the first line. Don't include multiple code blocks in one response. Do not ask users to copy and paste the result. Instead, use 'print' function for the output when relevant. Check the execution result returned by the user. If the result indicates there is an error, fix the error and output the code again. Suggest the full code instead of partial code or code changes. If the error can't be fixed or if the task is not solved even after the code is executed successfully, analyze the problem, revisit your assumption, collect additional info you need, and think of a different approach to try. When you find an answer, verify the answer carefully. Include verifiable evidence in your response if possible. Reply 'TERMINATE' in the end when everything is done.",
  };

  const assistantFlowSpec: IAgent = {
    type: "assistant",
    config: assistantConfig,
  };

  const workFlowConfig: IWorkflow = {
    name: "Default Agent Workflow",
    description: "Default Agent Workflow",
    sender: userProxyFlowSpec,
    receiver: assistantFlowSpec,
    type: "twoagents",
  };

  const groupChatAssistantConfig = Object.assign(
    {
      admin_name: "groupchat_assistant",
      messages: [],
      max_round: 10,
      speaker_selection_method: "auto",
      allow_repeat_speaker: false,
      description: "Group Chat Assistant",
    },
    assistantConfig
  );
  groupChatAssistantConfig.name = "groupchat_assistant";
  groupChatAssistantConfig.system_message =
    "You are a helpful assistant skilled at cordinating a group of other assistants to solve a task. ";

  const groupChatFlowSpec: IAgent = {
    type: "groupchat",
    config: groupChatAssistantConfig,
  };

  const groupChatWorkFlowConfig: IWorkflow = {
    name: "Default Group Workflow",
    description: "Default Group  Workflow",
    sender: userProxyFlowSpec,
    receiver: groupChatFlowSpec,
    type: "groupchat",
  };

  if (type === "twoagents") {
    return workFlowConfig;
  } else if (type === "groupchat") {
    return groupChatWorkFlowConfig;
  }
  return workFlowConfig;
};

export const getSampleSkill = () => {
  const content = `
from typing import List
import uuid
import requests  # to perform HTTP requests
from pathlib import Path

from openai import OpenAI


def generate_and_save_images(query: str, image_size: str = "1024x1024") -> List[str]:
    """
    Function to paint, draw or illustrate images based on the users query or request. Generates images from a given query using OpenAI's DALL-E model and saves them to disk.  Use the code below anytime there is a request to create an image.

    :param query: A natural language description of the image to be generated.
    :param image_size: The size of the image to be generated. (default is "1024x1024")
    :return: A list of filenames for the saved images.
    """

    client = OpenAI()  # Initialize the OpenAI client
    response = client.images.generate(model="dall-e-3", prompt=query, n=1, size=image_size)  # Generate images

    # List to store the file names of saved images
    saved_files = []

    # Check if the response is successful
    if response.data:
        for image_data in response.data:
            # Generate a random UUID as the file name
            file_name = str(uuid.uuid4()) + ".png"  # Assuming the image is a PNG
            file_path = Path(file_name)

            img_url = image_data.url
            img_response = requests.get(img_url)
            if img_response.status_code == 200:
                # Write the binary content to a file
                with open(file_path, "wb") as img_file:
                    img_file.write(img_response.content)
                    print(f"Image saved to {file_path}")
                    saved_files.append(str(file_path))
            else:
                print(f"Failed to download the image from {img_url}")
    else:
        print("No image data found in the response!")

    # Return the list of saved files
    return saved_files


# Example usage of the function:
# generate_and_save_images("A cute baby sea otter")
  `;

  const skill: ISkill = {
    name: "generate_images",
    description: "Generate and save images based on a user's query.",
    content: content,
  };

  return skill;
};

export const timeAgo = (
  dateString: string,
  returnFormatted: boolean = false
): string => {
  // if dateStr is empty, return empty string
  if (!dateString) {
    return "";
  }
  // Parse the date string into a Date object
  const timestamp = new Date(dateString);

  // Check for invalid date
  if (isNaN(timestamp.getTime())) {
    throw new Error("Invalid date string provided.");
  }

  // Get the current time
  const now = new Date();

  // Calculate the difference in milliseconds
  const timeDifference = now.getTime() - timestamp.getTime();

  // Convert time difference to minutes and hours
  const minutesAgo = Math.floor(timeDifference / (1000 * 60));
  const hoursAgo = Math.floor(minutesAgo / 60);

  // Format the date into a readable format e.g. "November 27, 2021, 3:45 PM"
  const options: Intl.DateTimeFormatOptions = {
    month: "long",
    day: "numeric",
    year: "numeric",
    hour: "numeric",
    minute: "numeric",
  };
  const formattedDate = timestamp.toLocaleDateString(undefined, options);

  if (returnFormatted) {
    return formattedDate;
  }

  // Determine the time difference string
  let timeAgoStr: string;
  if (minutesAgo < 1) {
    timeAgoStr = "just now";
  } else if (minutesAgo < 60) {
    // Less than an hour ago, display minutes
    timeAgoStr = `${minutesAgo} ${minutesAgo === 1 ? "minute" : "minutes"} ago`;
  } else if (hoursAgo < 24) {
    // Less than a day ago, display hours
    timeAgoStr = `${hoursAgo} ${hoursAgo === 1 ? "hour" : "hours"} ago`;
  } else {
    // More than a day ago, display the formatted date
    timeAgoStr = formattedDate;
  }

  // Return the final readable string
  return timeAgoStr;
};

export const examplePrompts = [
  {
    title: "Stock Price",
    prompt:
      "Plot a chart of NVDA and TESLA stock price for 2023. Save the result to a file named nvda_tesla.png",
  },
  {
    title: "Sine Wave",
    prompt:
      "Write a python script to plot a sine wave and save it to disc as a png file sine_wave.png",
  },
  {
    title: "Markdown",
    prompt:
      "List out the top 5 rivers in africa and their length and return that as a markdown table. Do not try to write any code, just write the table",
  },
  {
    title: "Paint",
    prompt:
      "paint a picture of a glass of ethiopian coffee, freshly brewed in a tall glass cup, on a table right in front of a lush green forest scenery",
  },
  {
    title: "Travel",
    prompt:
      "Plan a 2 day trip to hawaii. Limit to 3 activities per day, be as brief as possible!",
  },
];

export const fetchVersion = () => {
  const versionUrl = getServerUrl() + "/version";
  return fetch(versionUrl)
    .then((response) => response.json())
    .then((data) => {
      return data;
    })
    .catch((error) => {
      console.error("Error:", error);
      return null;
    });
};

/**
 * Recursively sanitizes JSON objects by replacing specific keys with a given value.
 * @param {JsonValue} data - The JSON data to be sanitized.
 * @param {string[]} keys - An array of keys to be replaced in the JSON object.
 * @param {string} replacement - The value to use as replacement for the specified keys.
 * @returns {JsonValue} - The sanitized JSON data.
 */
export const sanitizeConfig = (
  data: any,
  keys: string[] = ["api_key", "id", "created_at", "updated_at"]
): any => {
  if (Array.isArray(data)) {
    return data.map((item) => sanitizeConfig(item, keys));
  } else if (typeof data === "object" && data !== null) {
    Object.keys(data).forEach((key) => {
      if (keys.includes(key)) {
        delete data[key];
      } else {
        data[key] = sanitizeConfig(data[key], keys);
      }
    });
  }
  return data;
};

/**
 * Checks the input text against the regex '^[a-zA-Z0-9_-]{1,64}$' and returns an object with
 * status, message, and sanitizedText. Status is boolean indicating whether input text is valid,
 * message provides information about the outcome, and sanitizedText contains a valid version
 * of the input text or the original text if it was already valid.
 *
 * @param text - The input string to be checked and sanitized.
 * @returns An object containing a status, a message, and sanitizedText.
 */
export const checkAndSanitizeInput = (
  text: string
): { status: boolean; message: string; sanitizedText: string } => {
  // Create a regular expression pattern to match valid characters
  const regexPattern: RegExp = /^[a-zA-Z0-9_-]{1,64}$/;
  let status: boolean = true;
  let message: string;
  let sanitizedText: string;

  // Check if the input text matches the pattern
  if (regexPattern.test(text)) {
    // Text already adheres to the pattern
    message = `The text '${text}' is valid.`;
    sanitizedText = text;
  } else {
    // The text does not match; sanitize the input
    status = false;
    sanitizedText = text.replace(/[^a-zA-Z0-9_-]/g, "_").slice(0, 64);
    message = `'${text}' is invalid. Consider using '${sanitizedText}' instead.`;
  }

  return { status, message, sanitizedText };
};

export const isValidConfig = (
  jsonObj: any,
  templateObj: any,
  diffThreshold: number = 4
): {
  status: boolean;
  message: string;
} => {
  // Check if both parameters are indeed objects and not null
  if (
    typeof jsonObj !== "object" ||
    jsonObj === null ||
    Array.isArray(jsonObj) ||
    typeof templateObj !== "object" ||
    templateObj === null ||
    Array.isArray(templateObj)
  ) {
    return {
      status: false,
      message:
        "Invalid input: One or both parameters are not objects, or are null or arrays.",
    };
  }

  const jsonKeys = new Set(Object.keys(jsonObj));
  const templateKeys = new Set(Object.keys(templateObj));

  if (jsonKeys.size !== templateKeys.size) {
    if (Math.abs(jsonKeys.size - templateKeys.size) > diffThreshold) {
      return {
        status: false,
        message:
          "Configuration does not match template: Number of keys differ.",
      };
    }
  }

  for (const key of templateKeys) {
    if (!jsonKeys.has(key)) {
      return {
        status: false,
        message: `Configuration does not match template: Missing key '${key}' in configuration.`,
      };
    }

    // If the value is an object, recursively validate
    if (
      typeof templateObj[key] === "object" &&
      templateObj[key] !== null &&
      !Array.isArray(templateObj[key])
    ) {
      const result = isValidConfig(jsonObj[key], templateObj[key]);
      if (!result.status) {
        return {
          status: false,
          message: `Configuration error in nested key '${key}': ${result.message}`,
        };
      }
    }
  }

  return {
    status: true,
    message: "Configuration is valid.",
  };
};
